package com.health.common.utils;

import cn.hutool.core.convert.Convert;
import cn.hutool.jwt.Claims;
import com.auth0.jwt.JWT;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.interfaces.DecodedJWT;
import com.auth0.jwt.interfaces.JWTVerifier;
import com.health.common.constant.SecurityConstants;
import com.health.common.constant.TokenConstants;

import java.util.Map;

import static com.health.common.constant.SecurityConstants.DETAILS_USER_ID;
import static com.health.common.constant.UserConstant.LOGIN_TOKEN_KEY;

/**
 * Jwt工具类
 *
 * @author ruoyi
 */
public class JwtUtils
{
    public static String SECRET = TokenConstants.SECRET;

    private static final Algorithm ALGORITHM = Algorithm.HMAC256(SECRET);


    /**
     * 秘钥实例
     */
//    public static final SecretKey KEY = SECRET.length() < 32
//            ? Keys.secretKeyFor(SignatureAlgorithm.HS256)  // 生成一个安全的密钥
//            : Keys.hmacShaKeyFor(SECRET.getBytes());

    /**
     * 从数据声明生成令牌
     *
     * @param claims 数据声明
     * @return 令牌
     */
    public static String createToken(Map<String, Object> claims) {
        return JWT.create()
                .withPayload(claims)
                .sign(ALGORITHM);
    }

    /**
     * 从令牌中获取数据声明
     *
     * @param token 令牌
     * @return 数据声明
     */
//    public static Claims parseToken(String token)
//    {
//        return Jwts.parser().setSigningKey(KEY).build().parseClaimsJws(token).getBody();
//    }

    public static DecodedJWT parseToken(String token) {
        JWTVerifier verifier = JWT.require(ALGORITHM).build();
        return verifier.verify(token);
    }

    /**
     * 根据令牌获取用户标识
     *
     * @param token 令牌
     * @return 用户ID
     */
//    public static String getUserKey(String token)
//    {
//        Claims claims = parseToken(token);
//        return getValue(claims, SecurityConstants.USER_KEY);
//    }

    public static String getUserKey(String token) {
        DecodedJWT jwt = parseToken(token);
        return jwt.getClaim(SecurityConstants.USER_KEY).asString();
    }

    /**
     * 根据令牌获取用户标识
     *
     * @param claims 身份信息
     * @return 用户ID
     */
    public static String getUserKey(Claims claims)
    {
        return getValue(claims, SecurityConstants.USER_KEY);
    }

    /**
     * 根据令牌获取用户ID
     *
     * @param token 令牌
     * @return 用户ID
     */
//    public static String getUserId(String token)
//    {
//        Claims claims = parseToken(token);
//        return getValue(claims, SecurityConstants.DETAILS_USER_ID);
//    }
    public static String getUserId(String token) {
        DecodedJWT jwt = parseToken(token);
        return jwt.getClaim(SecurityConstants.DETAILS_USER_ID).asString();
    }

    /**
     * 根据身份信息获取用户ID
     *
     * @param claims 身份信息
     * @return 用户ID
     */
    public static String getUserId(Claims claims)
    {
        return getValue(claims, DETAILS_USER_ID);
    }

    /**
     * 根据令牌获取用户名
     *
     * @param token 令牌
     * @return 用户名
     */
//    public static String getUserName(String token)
//    {
//        Claims claims = parseToken(token);
//        return getValue(claims, SecurityConstants.DETAILS_USERNAME);
//    }
    public static String getUserName(String token) {
        DecodedJWT jwt = parseToken(token);
        return jwt.getClaim("username").asString();
    }

    /**
     * 根据身份信息获取用户名
     *
     * @param claims 身份信息
     * @return 用户名
     */
    public static String getUserName(Claims claims)
    {
        return getValue(claims, SecurityConstants.DETAILS_USERNAME);
    }

    /**
     * 根据身份信息获取键值
     *
     * @param claims 身份信息
     * @param key 键
     * @return 值
     */
    public static String getValue(Claims claims, String key)
    {
        return Convert.toStr(claims.getClaim(key), "");
    }

    public static String getTokenKey(String token)
    {
        return LOGIN_TOKEN_KEY + token;
    }
}
